/**
* Software License, Version 1.0
 *
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 *
 *
 *Redistribution and use in source and binary forms, with or without
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For
 * further information contact the Community Grids Lab at
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and
 * wherever such third-party acknowledgments normally appear.
 *
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering,
 * shall not be used to endorse or promote products derived from this software
 * without prior written permission from Indiana University.  For written
 * permission, please contact the Advanced Research and Technology Institute
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering,
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 *
 *
 * Indiana University provides no reassurances that the source code provided
 * does not infringe the patent or any other intellectual property rights of
 * any other entity.  Indiana University disclaims any liability to any
 * recipient for claims brought by any other entity based on infringement of
 * intellectual property rights or otherwise.
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS.
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS",
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.wsinfra.deployment.axis.wsr.proxy;

import java.math.BigInteger;

import javax.xml.namespace.QName;
import javax.xml.soap.SOAPMessage;

import org.apache.xmlbeans.XmlCursor;

import cgl.narada.wsinfra.WsMessageFlow;
import cgl.narada.wsinfra.exception.IncorrectExchangeException;
import cgl.narada.wsinfra.exception.MessageFlowException;
import cgl.narada.wsinfra.exception.ParsingException;
import cgl.narada.wsinfra.exception.ProcessingException;
import cgl.narada.wsinfra.exception.UnknownExchangeException;
import cgl.narada.wsinfra.exception.WsFaultException;
import cgl.narada.wsinfra.schemas.addressing.ActionDocument;
import cgl.narada.wsinfra.schemas.addressing.EndpointReferenceType;
import cgl.narada.wsinfra.schemas.addressing.FromDocument;
import cgl.narada.wsinfra.schemas.addressing.RelatesToDocument;
import cgl.narada.wsinfra.schemas.addressing.ToDocument;
import cgl.narada.wsinfra.schemas.soap.EnvelopeDocument;
import cgl.narada.wsinfra.schemas.wsr.RequestDocument;
import cgl.narada.wsinfra.util.SoapEnvelopeConversion;
import cgl.narada.wsinfra.util.SoapMarshaller;
import cgl.narada.wsinfra.util.SoapMessageAlteration;
import cgl.narada.wsinfra.wsa.AddressingHeaders;
import cgl.narada.wsinfra.wsa.ParseWsaHeaders;
import cgl.narada.wsinfra.wsa.WsaEnvelopeCreator;
import cgl.narada.wsinfra.wsa.WsaEprCreator;
import cgl.narada.wsinfra.wsa.WsaProcessingFactory;
import cgl.narada.wsinfra.wsr.WsrActions;
import cgl.narada.wsinfra.wsr.WsrNodeUtils;
import cgl.narada.wsinfra.wsr.WsrProcessingFactory;
/**
 * @author Damodar R Yemme
 * $DateApr 19, 2005$
 * $Revision 1.0$
 * This class will act as a Appliaction for Sink Node and will process
 * SOAPMessage sent by Sink and responds with Payload if any from Application.
 */
public class WsrTargetServiceProxy {
    private String moduleName = "WsrTargetServiceProxy";
    private WsaEprCreator eprCreator;
    private EndpointReferenceType sinkEpr, sourceEpr, targetServiceEpr;
  	private String sinkAddress;
  	private String sourceAddress;
  	private String targetServiceAddress;
  	private WsrNodeUtils wsrNodeUtils;
    private WsrTargetServiceProxyHelper wsrTargetServiceProxyHelper;
    private WsaEprCreator wsaEprCreator;
    private WsaEnvelopeCreator wsaEnvelopeCreator;
    private SoapEnvelopeConversion soapEnvelopeConversion;
    private WsrActions wsrActions;
    private SoapMessageAlteration soapMessageAlteration;
    private String methodName;
    private int tracker;
    /**
     * This Constructor class will initiates all services to process SOAPMessage
     * at Sink Node Application.
     *
     */
    public WsrTargetServiceProxy(){
        sourceAddress = WsrProxyServicesFactory.getSourceAddress();
        sinkAddress = WsrProxyServicesFactory.getSinkAddress();
        wsrTargetServiceProxyHelper = new WsrTargetServiceProxyHelper();
        targetServiceAddress =WsrProxyServicesFactory.getTargetServiceAddress();
        eprCreator = WsaProcessingFactory.getWsaEprCreator();
        wsrNodeUtils = WsrProcessingFactory.getWsrNodeUtils();

        sourceAddress = WsrProxyServicesFactory.getSourceAddress();
        sinkAddress = WsrProxyServicesFactory.getSinkAddress();
        targetServiceAddress =WsrProxyServicesFactory.getTargetServiceAddress();
        eprCreator = WsaProcessingFactory.getWsaEprCreator();
        sourceEpr = eprCreator.createEpr(sourceAddress);
        sinkEpr = eprCreator.createEpr(sinkAddress);
        targetServiceEpr = eprCreator.createEpr(targetServiceAddress);
        wsrNodeUtils = WsrProcessingFactory.getWsrNodeUtils();
        methodName = WsrProxyServicesFactory.getMethodName();
        wsrActions = WsrActions.getInstance();
        wsaEprCreator = WsaProcessingFactory.getWsaEprCreator();
        wsaEnvelopeCreator = WsaProcessingFactory.getWsaEnvelopeCreator();
        soapEnvelopeConversion = SoapEnvelopeConversion.getInstance();
        soapMessageAlteration = SoapMessageAlteration.getInstance();
    }
    /**
     * This will do the web service for Source in AXIS
     * @param req
     * @param res
     * @throws WsFaultException
     * @throws ParsingException
     * @throws ProcessingException
     */
    public void wsInfraProcessMethod(org.apache.axis.message.SOAPEnvelope req,
                                     org.apache.axis.message.SOAPEnvelope res)
    throws WsFaultException, ParsingException, ProcessingException {
      try {

        SOAPMessage soapMessage = SoapMarshaller.unmarshallSoapMessage(
            req.toString().getBytes());
        System.out.println("*********************************************\n");
  	    System.out.println(moduleName + "Envelope in Proxy TS Axis \n\n" +req);
  	    System.out.println("*********************************************");
            SoapEnvelopeConversion soapEnvelopeConversion=SoapEnvelopeConversion
                    .getInstance();
            EnvelopeDocument envelopeDocument = soapEnvelopeConversion
                    .getEnvelopeDocument(soapMessage);

	    QName qName = new QName("http://docs.oasis-open.org/wss/2004/01/oasis-"+
	    		"200401-wss-wssecurity-secext-1.0.xsd","Security");      
      	    SoapMessageAlteration soapMessageAlteration = 
      	    						SoapMessageAlteration.getInstance();
      	    soapMessageAlteration.removeFromSoapHeader(envelopeDocument,qName); 
			
            String from = null;
            String to = null;
            ToDocument toDocument = null;
            FromDocument fromDocument = null;
            AddressingHeaders addressingHeaders = null;
            ParseWsaHeaders parseWsaHeaders = WsaProcessingFactory
                    .getParseWsaHeaders();

            try {
                toDocument = parseWsaHeaders.getTo(envelopeDocument);
                fromDocument = parseWsaHeaders.getFrom(envelopeDocument);
                addressingHeaders = parseWsaHeaders
                        .getAddressingHeaders(envelopeDocument);
            } catch (ParsingException parseEx) {
                // Ignore this exception.
                System.out.println("Problems parsing the envelope"
                        + parseEx.toString());
            }
            String action = null;
            ActionDocument actionDocument = null;
            try {
                actionDocument = addressingHeaders.getAction();
                if (actionDocument == null) {
                    String reason = moduleName
                            + "Action missing. Processor only deals with "
                            + "WS-Eventing exchanges.";
                }

                if (actionDocument != null) {
                    action = actionDocument.getAction().getStringValue();
                }
            } catch (Exception e) {
                System.out.println(moduleName
                        + "Exception in Processing action " + e);
            }

            if (toDocument == null) {
                to = "";
            } else {
                to = toDocument.getTo().getStringValue();

            }
            if (to.equals("")) {
                System.out.println("**Received SOAPMessage from Application**");
                processMessage(envelopeDocument,WsMessageFlow.FROM_APPLICATION);
            } else {
                System.out.println("***Received SOAPMessage from Network*****");
                processMessage(envelopeDocument, WsMessageFlow.FROM_NETWORK);
            }

        } catch (UnknownExchangeException ex) {
            System.out
                    .println(moduleName + " UnknownExchangeException : " + ex);
        } catch (IncorrectExchangeException ex) {
            System.out.println(moduleName + " IncorrectExchangeException : "
                    + ex);
        } catch (MessageFlowException ex) {
            System.out.println(moduleName + " MessageFlowException : " + ex);
        } catch (Exception ex) {
            System.out.println(moduleName + " Exception in processMessage() "
                    + ex);
        }
    }

    /**
     * This method returns a boolean which indicates whether further processing
     * should continue or if it should stop. A return value of <i>true </i>
     * indicates that processing should continue; while <i>false </i> indicates
     * that processing should stop. Note that it is the filter-pipeline which is
     * responsible for stopping this processing.
     */
    public boolean processMessage(EnvelopeDocument envelopeDocument,
            int direction) throws UnknownExchangeException,
            IncorrectExchangeException, MessageFlowException,
            ProcessingException,WsFaultException {
        String from = "the APPLICATION.";
        if (direction == WsMessageFlow.FROM_NETWORK) {
            from = "the Network.";
        }
        RequestDocument requestDocument =
            wsrNodeUtils.getRequestDocument(envelopeDocument);
        String groupId=requestDocument.getRequest().getMessageId().getGroupId();
        BigInteger seqNumber = requestDocument.getRequest().getMessageId()
				.getSequenceNum().getNumber();
        System.out.println(moduleName
                + "Processing SOAP EnvelopeDocument received from " + from);
        //Set the FROM: Target Service..
        String fromAddress = WsrProxyServicesFactory.getTargetServiceAddress();
        
        
        String fromToToAddress = null;       
        
        FromDocument fromDocument = null;
        AddressingHeaders addressingHeaders = null;
        ParseWsaHeaders parseWsaHeaders = WsaProcessingFactory
                .getParseWsaHeaders();

        try {
            fromDocument = parseWsaHeaders.getFrom(envelopeDocument);
            addressingHeaders = parseWsaHeaders
                    .getAddressingHeaders(envelopeDocument);
        } catch (ParsingException parseEx) {
            // Ignore this exception.
            System.out.println("Problems parsing the envelope"+moduleName+"\n");
            parseEx.printStackTrace();
        }        
        fromToToAddress = fromDocument.getFrom().getAddress().getStringValue();	
        String toAddress = fromToToAddress;//sinkAddress;
               
        //Set TO :Source address
        //String toAddress = sourceAddress;
        SOAPMessage soapMessage =
        			createSOAPMessage(fromAddress,toAddress,groupId, seqNumber);
        wsrTargetServiceProxyHelper.enrouteToNetwork(soapMessage);
        return true;
    }
    /**
     * This method will creates SOAPMessage based on given parameters.
     * @param from address
     * @param to address
     * @param groupId in the received message for destination.
     * @param seqNumber
     * @return javax.xml.soap.SOAPMessage
     */
    public SOAPMessage createSOAPMessage(String from, String to,
    								String groupId, BigInteger seqNumber) {

        EndpointReferenceType sourceEpr = wsaEprCreator.createEpr(from);

        EndpointReferenceType sinkEpr = wsaEprCreator.createEpr(to);

        FromDocument fromDocument = FromDocument.Factory.newInstance();
        fromDocument.setFrom(sourceEpr);

        ActionDocument action = ActionDocument.Factory.newInstance();
        action.addNewAction().setStringValue(wsrActions.getProcessResponse());
        RelatesToDocument relatesTo = null;

        EnvelopeDocument envelopeDocument = wsaEnvelopeCreator
                .createSoapEnvelope(sinkEpr, fromDocument, action, relatesTo);

        QName qname = new QName("http://www.naradabrokering.org",
                "Application-Content");
        XmlCursor bodyCursor = envelopeDocument.getEnvelope().getBody()
                .newCursor();
        tracker++;
        String value = "PAYLOAD FROM APPLICATION for GroupID:" + groupId +
												" Sequence Number: "+seqNumber;
        bodyCursor.toNextToken();
        bodyCursor.insertElementWithText(qname, value);
        bodyCursor.dispose();
        System.out.println(envelopeDocument.toString());
        /*
        if (groupId != null) {
            QName qName = WsrQNames.getInstance().getGroupId();
            boolean added = soapMessageAlteration.addToSoapHeader(
                    envelopeDocument, qName, groupId);
            if (added) {
                System.out.println(moduleName + "GroupId added successfully");
            }
        }
        */
        SOAPMessage soapMessage = null;
        try {
            soapMessage = soapEnvelopeConversion
                    .getSOAPMessage(envelopeDocument);
        } catch (ProcessingException ex) {
            String errorReport ="Problems converting to javax.xml.SOAPMessage\n"
                    + ex.toString();
            System.out.println(errorReport);
        }
        return soapMessage;

    }
}

